package com.springboot.demo.config;

import java.sql.Types;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.internal.types.JavaTypeResolverDefaultImpl;

public class JavaTypeCustomResolver extends JavaTypeResolverDefaultImpl {

	public JavaTypeCustomResolver() {
		super();
		//$NON-NLS-1$
		typeMap.put(Types.TINYINT, new JdbcTypeInformation("TINYINT",
                new FullyQualifiedJavaType(Integer.class.getName())));
		//$NON-NLS-1$
		typeMap.put(Types.SMALLINT, new JdbcTypeInformation("SMALLINT",
                new FullyQualifiedJavaType(Integer.class.getName())));
	}

}
